package com.wteam.dragon.system.exception.handler;

import cn.hutool.core.lang.Dict;
import com.wteam.dragon.system.apps.security.utils.ThrowableUtil;
import com.wteam.dragon.system.basebean.bean.ResultMessage;
import com.wteam.dragon.system.exception.BadRequestException;
import com.wteam.dragon.system.exception.EntityExistException;
import com.wteam.dragon.system.exception.EntityNotFoundException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import static org.springframework.http.HttpStatus.*;

/**
 * 自定义异常
 * @Author 陈晓辉
 * @Date 2020/5/29 17:11
 */
@Slf4j
@RestControllerAdvice
public class GlobalExceptionHandler {
    /**
     * 最大自定义错误信息长度
     */
    private final static Integer MAX_ERROR_MESSAGE_LENGTH = 30;

    /**
     * 处理所有不可知的异常
     */
    @ExceptionHandler(Throwable.class)
    public ResultMessage handleException(Throwable e) {
        // 打印堆栈信息
        log.error(ThrowableUtil.getStackTrace(e));
        Dict dict = Dict.create().set("具体的原因是", e.getStackTrace());
        if(e.getMessage().length() > MAX_ERROR_MESSAGE_LENGTH){
            return new ResultMessage(BAD_REQUEST.value(), "未知的错误异常，请联系管理员解决", dict);
        }
        return new ResultMessage(BAD_REQUEST.value(), e.getMessage(), dict);
    }

    /**
     * 处理 接口无权访问异常AccessDeniedException
     */
    @ExceptionHandler(AccessDeniedException.class)
    public ResultMessage handleAccessDeniedException(AccessDeniedException e) {
        // 打印堆栈信息
        log.error(ThrowableUtil.getStackTrace(e));
        return new ResultMessage(FORBIDDEN.value(), e.getMessage());
    }

    /**
     * 处理自定义异常
     */
    @ExceptionHandler(value = BadRequestException.class)
    public ResultMessage badRequestException(BadRequestException e) {
        // 打印堆栈信息
        log.error(ThrowableUtil.getStackTrace(e));
        Dict dict = Dict.create().set("具体的原因是", e.getStackTrace());
        return new ResultMessage(e.getStatus(), e.getMessage(), dict);
    }

    /**
     * 处理 EntityExist
     */
    @ExceptionHandler(value = EntityExistException.class)
    public ResultMessage entityExistException(EntityExistException e) {
        // 打印堆栈信息
        log.error(ThrowableUtil.getStackTrace(e));
        return new ResultMessage(BAD_REQUEST.value(), e.getMessage());
    }

    /**
     * 处理 EntityNotFound
     */
    @ExceptionHandler(value = EntityNotFoundException.class)
    public ResultMessage entityNotFoundException(EntityNotFoundException e) {
        // 打印堆栈信息
        log.error(ThrowableUtil.getStackTrace(e));
        return new ResultMessage(NOT_FOUND.value(), e.getMessage());
    }

    /**
     * 处理所有接口数据验证异常
     */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    public ResultMessage handleMethodArgumentNotValidException(MethodArgumentNotValidException e) {
        // 打印堆栈信息
        log.error(ThrowableUtil.getStackTrace(e));
        String[] str = e.getBindingResult().getAllErrors().get(0).getCodes()[1].split("\\.");
        StringBuffer msg = new StringBuffer(str[1] + ":");
        msg.append(e.getBindingResult().getAllErrors().get(0).getDefaultMessage());
        return new ResultMessage(BAD_REQUEST.value(), msg.toString());
    }
}
